from dataclasses import dataclass

from keras.optimizers import Adam
from keras.optimizers.optimizer_v2 import optimizer_v2


@dataclass
class ModelConfig:
    def __init__(self):
        pass

    IMAGE_HEIGHT: int = 128
    IMAGE_WIDTH: int = 128
    EPOCHS: int = 10
    LEARNING_RATE: float = 0.001
    DECAY: float = 0.001
    COLOR_CHANNELS: int = 3
    OPTIMIZER = Adam
